#!/bin/bash

if [[ $# -ne 2 ]]; then
    echo "The format is as follows:"
    echo "bash fftw-3.3.10.sh buildpath installpath"
    echo ":param buildpath: 应用构建绝对路径"
    echo ":param installpath: 应用安装绝对路径"
    exit 1
fi

[[ ! "$1" =~ ^/.* || ! -d "$1" || "$1" = "/" ]] &&  echo "请输入正确的构建路径" && exit 1
[[ ! "$2" =~ ^/.* || ! -d "$2" || "$2" = "/" ]] &&  echo "请输入正确的安装路径" && exit 1
if [[ "$1" == "$2" ]]; then echo "构建路径与安装路径不能相同，请输入正确的路径"; exit 1; fi

buildpath=$1
installpath=$2
[[ "$buildpath" =~ .*/$ ]] && buildpath=${buildpath%/*}
[[ "$installpath" =~ .*/$ ]] && installpath=${installpath%/*}
set -e

download_pkg()
{
    wget -t 40 -c $1 -O $buildpath/fftw-3.3.10.tar.gz
    tar xf $buildpath/fftw-3.3.10.tar.gz -C $buildpath
    #rm -rf $buildpath/fftw-3.3.10.tar.gz
}

download_pkg https://fftw.org/pub/fftw/fftw-3.3.10.tar.gz
cd $buildpath/fftw-3.3.10

./configure CC=gcc MPICC=mpicc F77="gfortran -fallow-argument-mismatch" --prefix=$installpath/fftw-3.3.10 --enable-mpi --enable-openmp --enable-threads --enable-shared=yes --enable-static=yes --enable-neon --enable-armv8-pmccntr-el0 --enable-armv8-cntvct-el0
make -j$(nproc)
make install
./configure CC=gcc MPICC=mpicc F77="gfortran -fallow-argument-mismatch" --prefix=$installpath/fftw-3.3.10 --enable-mpi --enable-openmp --enable-threads --enable-float --enable-shared=yes --enable-static=yes --enable-neon --enable-armv8-pmccntr-el0 --enable-armv8-cntvct-el0
make clean
make -j$(nproc)
make install

cat>"$installpath/fftw-3.3.10/fftw_modulefiles"<<EOF
#%Module1.0
conflict fftw
variable modfile [file normalize [info script]]
proc getModulefileDir {} {
    variable modfile
    set modfile_path [file dirname \$modfile]
    return \$modfile_path
}
set fftw [getModulefileDir]
setenv FFTW \$fftw
prepend-path PATH \$fftw/bin
prepend-path LD_LIBRARY_PATH \$fftw/lib
prepend-path INCLUDE \$fftw/include
EOF

#rm -rf $buildpath/fftw-3.3.10
